How to write Flask API in Python